# -*- coding: utf-8 -*-
"""
@Time    : 2025/3/21 17:05 
@Author  : ZhangShenao 
@File    : lang_detect.py 
@Desc    : 
"""

import os

import dotenv
from openai import OpenAI

# 加载环境变量
dotenv.load_dotenv()

# 创建OpenAI客户端
client = OpenAI(base_url=os.getenv("OPENAI_API_BASE"))

prompt = """
【任务描述】
请帮助我完成多语种的识别任务。
特别注意：如果文本中包含多种语言，请按照语言在文本中的占比，由高到低返回。

【参考示例】
"你好 世界" -> 中文
"hello world" -> 英文
"你好 世界 今天天气真不错 hello world こんにちは" -> 中文|英文|日语

【具体任务】
"{text}" -> 
"""

completion = client.chat.completions.create(
    model="gpt-4o-mini",
    messages=[
        {"role": "system", "content": "你是一个精通多种语言的机器翻译专家，擅长多语言的翻译和识别工作。"},
        {"role": "user", "content": prompt.format(
            text="sonrio con aparente felicidad,  aunque por dentro me returzo de la incomodidad, por la mirada de Mor...")}
    ]
)

print(completion.choices[0].message.content)
